package com.wenx;

public class SetTreeData {

    public static void insert(TreeNode treeNode, int data) {

        //本结点为空或者相等，插入数据
        if (treeNode.getData() == null || treeNode.getData() == data) {
            treeNode.setData(data);
        }

        //向左子树插入数据
        if(treeNode.getData() > data) {
            //向左子树插入数据
            if(treeNode.getLeft() == null) {
                treeNode.setLeft(new TreeNode());
                treeNode.getLeft().setData(data);
            } else {
                //向左下寻找节点
                insert(treeNode.getLeft(),data);
            }
        }

        //向右子树插入数据
        if(treeNode.getData() < data) {
            //向右子树插入数据
            if(treeNode.getRight() == null) {
                treeNode.setRight(new TreeNode());
                treeNode.getRight().setData(data);
            } else {
                //向右下寻找节点
                insert(treeNode.getRight(),data);
            }
        }

    }

}
